package cn.zzf.leetcode;

import java.util.Arrays;

/**
 * <pre>{@code
 * 给你一个长度为 n 的整数数组 nums 和 一个目标值 target。请你从 nums 中选出三个整数，使它们的和与 target 最接近。
 * 返回这三个数的和。假定每组输入只存在恰好一个解。
 * 示例 1：
 * 输入：nums = [-1,2,1,-4], target = 1
 * 输出：2
 * 解释：与 target 最接近的和是 2 (-1 + 2 + 1 = 2)。
 *
 * 示例 2：
 * 输入：nums = [0,0,0], target = 1
 * 输出：0
 * 解释：与 target 最接近的和是 0（0 + 0 + 0 = 0）。
 *
 * 提示：
 * 3 <= nums.length <= 1000
 * -1000 <= nums[i] <= 1000
 * -104 <= target <= 104
 * }</pre>
 *
 * @author zzf
 * @date 2025-06-12 11:38
 */
public class T0016_ThreeSumClosest {

    public static int threeSumClosest(int[] nums, int target) {
        int idx = 0, left, right;
        Arrays.sort(nums);
        int res = nums[0] + nums[1] + nums[2];
        // 小于最小值就没必要算了
        if (target < res) return res;
        boolean stop = false;
        while (idx < nums.length - 2) {
            left = idx + 1;
            right = nums.length - 1;
            // 大于当前最大值就直接下一位。
            int max = nums[idx] + nums[nums.length - 2] + nums[nums.length - 1];
            if (target > max) {
                res = max;
                idx++;
                continue;
            }
            while (left < right) {

                int sum = nums[idx] + nums[left] + nums[right];
                if (sum == target) {
                    return target;
                }
                int currentDiff = Math.abs(sum - target);
                int lastDiff = Math.abs(res - target);
                if (currentDiff < lastDiff) {
                    res = sum;
                }
                if (sum > target) {

                    do right--;
                    while (left != right && nums[right] == nums[right + 1]);

                } else {
                    do left++;
                    while (left != right && nums[left] == nums[left - 1]);
                }
                // 如果target<0, 且nums[idx] + nums[left] >= 0，说明后面的都不可能比它小了。反之同理
                if ((target < 0 && nums[idx] + nums[left] >= 0)
                        || (target > 0 && nums[right] <= 0)) {
                    stop = true;
                    break;
                }
            }
            if (stop) break;
            do idx++;
            while (idx < nums.length - 2 && nums[idx] == nums[idx + 1]);
        }
        return res;
    }

    public static void main(String[] args) {
        System.out.println(threeSumClosest(new int[]{-1, 2, 1, -4}, 1));
        System.out.println(threeSumClosest(new int[]{0, 0, 0}, 1));
    }

}
